"""
数据模型包初始化
"""

from .ai_web_agent import (
    AIWebAgent, AIWebAgentCreate, AIWebAgentUpdate, AIWebAgentResponse,
    AIWebAgentListResponse, AIWebAgentQueryParams, AIWebAgentStatusUpdate,
    AIWebAgentVersionUpdate, AgentStatus
)
from .ai_web_dialog import (
    AIWebDialog, AIWebDialogCreate, AIWebDialogUpdate, AIWebDialogResponse,
    AIWebDialogListResponse, AIWebDialogQueryParams, ButtonActionRequest,
    InteractiveMode, ButtonType, DialogSize, ButtonConfig
)
from .ai_web_memory_his import (
    AIWebMemoryHis, AIWebMemoryHisCreate, AIWebMemoryHisUpdate, AIWebMemoryHisResponse,
    AIWebMemoryHisListResponse, AIWebMemoryHisQueryParams, AIWebMemoryHisStatusUpdate,
    AIWebMemoryHisImportanceUpdate, AIWebMemoryHisBatchUpdate, AIWebMemoryHisStatistics,
    MemoryStatus, ImportanceLevel
)
from .ai_web_scene import (
    AIWebScene, AIWebSceneCreate, AIWebSceneUpdate,
    AIWebSceneResponse, AIWebSceneTreeResponse,
    AIWebSceneQueryParams, SceneStatus
)
from .ai_web_tools import (
    AIWebTools, AIWebToolsCreate, AIWebToolsUpdate, AIWebToolsResponse,
    AIWebToolsListResponse, AIWebToolsQueryParams, AIWebToolsStatusUpdate,
    AIWebToolsDetailResponse, ToolsStatus, ToolParameter, SingleToolModel
)

__all__ = [
    # 场景相关
    "AIWebScene",
    "AIWebSceneCreate",
    "AIWebSceneUpdate",
    "AIWebSceneResponse",
    "AIWebSceneTreeResponse",
    "AIWebSceneQueryParams",
    "SceneStatus",

    # 对话框相关
    "AIWebDialog",
    "AIWebDialogCreate",
    "AIWebDialogUpdate",
    "AIWebDialogResponse",
    "AIWebDialogListResponse",
    "AIWebDialogQueryParams",
    "ButtonActionRequest",
    "InteractiveMode",
    "ButtonType",
    "DialogSize",
    "ButtonConfig",

    # 智能体相关
    "AIWebAgent",
    "AIWebAgentCreate",
    "AIWebAgentUpdate",
    "AIWebAgentResponse",
    "AIWebAgentListResponse",
    "AIWebAgentQueryParams",
    "AIWebAgentStatusUpdate",
    "AIWebAgentVersionUpdate",
    "AgentStatus",

    # 工具集相关
    "AIWebTools",
    "AIWebToolsCreate",
    "AIWebToolsUpdate",
    "AIWebToolsResponse",
    "AIWebToolsListResponse",
    "AIWebToolsQueryParams",
    "AIWebToolsStatusUpdate",
    "AIWebToolsDetailResponse",
    "ToolsStatus",
    "ToolParameter",
    "SingleToolModel",

    # 历史消息记忆管理相关
    "AIWebMemoryHis",
    "AIWebMemoryHisCreate",
    "AIWebMemoryHisUpdate",
    "AIWebMemoryHisResponse",
    "AIWebMemoryHisListResponse",
    "AIWebMemoryHisQueryParams",
    "AIWebMemoryHisStatusUpdate",
    "AIWebMemoryHisImportanceUpdate",
    "AIWebMemoryHisBatchUpdate",
    "AIWebMemoryHisStatistics",
    "MemoryStatus",
    "ImportanceLevel"
]
